package com.strelitzia.listener;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.strelitzia.entity.ChatContext;
import com.strelitzia.service.UserService;
import com.strelitzia.utils.FileUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

@Component
@Slf4j
@ServerEndpoint(value = "/socket/user/{userId}")
public class UserSocketListener {
    private static UserService userService;
    @Autowired
    public void setUserService(UserService userService2) {
        userService = userService2;
    }

    public static final Map<String, Session> sessionMap = new ConcurrentHashMap<>();

    @OnOpen
    public void onOpen(Session session, @PathParam("userId") String userId) {
        if (sessionMap.get(userId) != null) {
            Session outSession = sessionMap.get(userId);
            log.info("userId={} 重新登陆", userId);
            sendMessage("您的账号在别处登录", outSession);
        } else {
            log.info("有新用户加入，userId={}", userId);
        }
        sessionMap.put(userId, session);
        log.info("当前在线人数{}", sessionMap.size());
    }

    /**
     * 连接关闭调用的方法
     */
    @OnClose
    public void onClose(Session session, @PathParam("userId") String userId) {
        if (sessionMap.get(userId).getId().equals(session.getId())) {
            sessionMap.remove(userId);
            userService.offline(Integer.parseInt(userId));
            log.info("userId={}已下线, 当前在线人数为：{}", userId, sessionMap.size());
        }

    }

    /**
     * 收到客户端消息后调用的方法
     * 后台收到客户端发送过来的消息
     * onMessage 是一个消息的中转站
     * 接受 浏览器端 socket.send 发送过来的 json数据
     *
     * @param message 客户端发送过来的消息
     */
    @OnMessage
    public void onMessage(String message, Session session, @PathParam("userId") String userId) throws ParseException {
        log.info("服务端收到用户userId={}的消息:{}", userId, message);
        ChatContext chatContext = JSON.parseObject(message, ChatContext.class);
        FileUtil.saveChatHistory(Integer.parseInt(userId), chatContext.getUid(), chatContext);
        if (sessionMap.get(chatContext.getUid().toString()) != null) {
            Session toSession = sessionMap.get(chatContext.getUid().toString());
            JSONObject jsonObj = new JSONObject();
            jsonObj.put("uid", userId);
            jsonObj.put("friend", chatContext.getUid());
            jsonObj.put("context", chatContext.getContext());
            String time = chatContext.getTime();
            SimpleDateFormat inputFormat = new SimpleDateFormat("yyyyMMddHHmm"); // 定义输入日期字符串的格式
            SimpleDateFormat outputFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm"); // 定义输出日期字符串的格式
            Date date = inputFormat.parse(time); // 将输入的日期字符串解析成Date对象
            String formattedDate = outputFormat.format(date); // 将Date对象按指定格式进行格式化
            jsonObj.put("time", formattedDate);
            sendMessage(jsonObj.toJSONString(), toSession);
        }

    }

    @OnError
    public void onError(Session session, Throwable error) {
        log.error("发生错误");
        error.printStackTrace();
    }

    /**
     * 服务端发送消息给客户端
     */
    private void sendMessage(String message, Session toSession) {
        try {
            log.info("服务端给客户端[{}]发送消息{}", toSession.getId(), message);
            toSession.getBasicRemote().sendText(message);
        } catch (Exception e) {
            log.error("服务端发送消息给客户端失败", e);
        }
    }

    /**
     * 服务端发送消息给所有客户端
     */
    private void sendAllMessage(String message) {
        try {
            for (Session session : sessionMap.values()) {
                log.info("服务端给客户端[{}]发送消息{}", session.getId(), message);
                session.getBasicRemote().sendText(message);
            }
        } catch (Exception e) {
            log.error("服务端发送消息给客户端失败", e);
        }
    }


}
